tensor([[0.3500, 0.3500, 0.3500, 0.3500, 0.3500],
        [0.6500, 0.6500, 0.6500, 0.6500, 0.6500]], grad_fn=<RoundBackward1>)